#ifndef DISPLAY_H
#define DISPLAY_H

#include <lvgl.h>
#include <esp_timer.h>
#include <esp_log.h>
#include <esp_pm.h>

#include <string>

/**
 * @brief 显示字体结构体
 * 包含文本、图标和表情符号的字体
 */
struct DisplayFonts {
    const lv_font_t* text_font = nullptr;   // 文本字体
    const lv_font_t* icon_font = nullptr;   // 图标字体
    const lv_font_t* emoji_font = nullptr;  // 表情符号字体
};

/**
 * @brief 显示控制基类
 * 负责管理和控制显示界面的各种元素
 */
class Display {
public:
    Display();
    virtual ~Display();

    /**
     * @brief 设置状态文本
     * @param status 状态文本
     */
    virtual void SetStatus(const char* status);
    
    /**
     * @brief 显示通知信息
     * @param notification 通知内容
     * @param duration_ms 显示持续时间(毫秒)
     */
    virtual void ShowNotification(const char* notification, int duration_ms = 3000);
    
    /**
     * @brief 显示通知信息(std::string版本)
     * @param notification 通知内容
     * @param duration_ms 显示持续时间(毫秒)
     */
    virtual void ShowNotification(const std::string &notification, int duration_ms = 3000);
    
    /**
     * @brief 设置表情
     * @param emotion 表情文本
     */
    virtual void SetEmotion(const char* emotion);
    
    /**
     * @brief 设置聊天消息
     * @param role 角色
     * @param content 消息内容
     */
    virtual void SetChatMessage(const char* role, const char* content);
    
    /**
     * @brief 设置图标
     * @param icon 图标文本
     */
    virtual void SetIcon(const char* icon);
    
    /**
     * @brief 设置主题
     * @param theme_name 主题名称
     */
    virtual void SetTheme(const std::string& theme_name);
    
    /**
     * @brief 获取当前主题
     * @return 当前主题名称
     */
    virtual std::string GetTheme() { return current_theme_name_; }

    /**
     * @brief 获取显示宽度
     * @return 宽度像素值
     */
    inline int width() const { return width_; }
    
    /**
     * @brief 获取显示高度
     * @return 高度像素值
     */
    inline int height() const { return height_; }

protected:
    int width_ = 0;                         // 显示宽度
    int height_ = 0;                        // 显示高度
    
    esp_pm_lock_handle_t pm_lock_ = nullptr; // 电源管理锁
    lv_display_t *display_ = nullptr;        // LVGL显示对象

    lv_obj_t *emotion_label_ = nullptr;      // 表情标签
    lv_obj_t *network_label_ = nullptr;      // 网络状态标签
    lv_obj_t *status_label_ = nullptr;       // 状态标签
    lv_obj_t *notification_label_ = nullptr; // 通知标签
    lv_obj_t *mute_label_ = nullptr;         // 静音标签
    lv_obj_t *battery_label_ = nullptr;      // 电池标签
    lv_obj_t* chat_message_label_ = nullptr; // 聊天消息标签
    lv_obj_t* low_battery_popup_ = nullptr;  // 低电量弹窗
    lv_obj_t* low_battery_label_ = nullptr;  // 低电量标签
    
    const char* battery_icon_ = nullptr;     // 电池图标
    const char* network_icon_ = nullptr;     // 网络图标
    bool muted_ = false;                     // 静音状态
    std::string current_theme_name_;         // 当前主题名称

    esp_timer_handle_t notification_timer_ = nullptr; // 通知计时器
    esp_timer_handle_t update_timer_ = nullptr;       // 更新计时器

    friend class DisplayLockGuard;
    /**
     * @brief 锁定显示
     * @param timeout_ms 超时时间(毫秒)
     * @return 是否成功锁定
     */
    virtual bool Lock(int timeout_ms = 0) = 0;
    
    /**
     * @brief 解锁显示
     */
    virtual void Unlock() = 0;

    /**
     * @brief 更新显示内容
     */
    virtual void Update();
};

/**
 * @brief 显示锁保护类
 * 使用RAII模式自动管理显示锁
 */
class DisplayLockGuard {
public:
    /**
     * @brief 构造函数，锁定显示
     * @param display 显示对象指针
     */
    DisplayLockGuard(Display *display) : display_(display) {
        if (!display_->Lock(3000)) {
            ESP_LOGE("Display", "Failed to lock display");
        }
    }
    
    /**
     * @brief 析构函数，解锁显示
     */
    ~DisplayLockGuard() {
        display_->Unlock();
    }

private:
    Display *display_;  // 显示对象指针
};

/**
 * @brief 空显示类
 * 用于不需要实际显示时的占位实现
 */
class NoDisplay : public Display {
private:
    /**
     * @brief 锁定显示(空实现)
     * @param timeout_ms 超时时间(毫秒)
     * @return 总是返回true
     */
    virtual bool Lock(int timeout_ms = 0) override {
        return true;
    }
    
    /**
     * @brief 解锁显示(空实现)
     */
    virtual void Unlock() override {}
};

#endif
